package com.peng.controller;

import cn.hutool.core.date.DateUtil;
import com.peng.entities.LawPrompt;
import com.peng.service.LawAssistant;
import dev.langchain4j.model.input.Prompt;
import dev.langchain4j.model.input.PromptTemplate;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.Map;
/**
 * @author love_ovo
 * @ClassName ChatLanguageModelController.java
 * @createTime 2025年05月13日 17:33:00
 */
@RestController
@Slf4j
public class ChatLanguageModelController {
    @Resource
    private LawAssistant lawAssistant;

    @GetMapping(value = "/chatprompt/test1")
    public String test1() {
        String chat = lawAssistant.chat("什么是知识产权？");
        System.out.println(chat);

        String chat2 = lawAssistant.chat("什么是java？");
        System.out.println(chat2);

        String chat3 = lawAssistant.chat("介绍下西瓜和芒果");
        System.out.println(chat3);

        String chat4 = lawAssistant.chat("飞机发动机原理");
        System.out.println(chat4);

        return "success : "+ DateUtil.now() + "\t"+chat +"\n\n\n"+chat2;
    }

    /**
     * TRIPS协议（与贸易有关的知识产权协议）：
     * 这是世界贸易组织（WTO）成员间的一个重要协议，
     * 它规定了最低标准的知识产权保护要求，并适用于所有WTO成员。
     * @return
     */
    @GetMapping(value = "/chatprompt/test2")
    public String test2() {
        LawPrompt prompt = new LawPrompt();

        prompt.setLegal("知识产权");
        prompt.setQuestion("TRIPS协议?");

        String chat = lawAssistant.chat(prompt);

        System.out.println(chat);

        return "success : "+ DateUtil.now() + "\t"+chat;
    }

    //最石板的写法，不推荐
    /**
     * 这是一个使用PromptTemplate的示例。
     * 它创建了一个模板，其中包含一个占位符{{it}}，
     * 然后使用apply方法将实际值"知识产权"填充到模板中。*/
    @GetMapping(value = "/chatprompt/test3")
    public String test3() {
        // 默认 form 构造使用 it 属性作为默认占位符
        PromptTemplate template = PromptTemplate.from("请解释中国法律中的'{{it}}'概念v1。");
        Prompt prompt = template.apply("知识产权");
        System.out.println(prompt.text()); // 输出: 请解释中国法律中的'知识产权'概念。

        // apply 方法接受 Map 作为参数
        PromptTemplate template2 = PromptTemplate.from("请解释中国法律中的'{{legal1}}'概念v2。");
        Prompt prompt2 = template2.apply(Map.of("legal1", "知识产权"));
        System.out.println(prompt2.text());// 输出: 请解释中国法律中的'知识产权'概念。

        return "test3 success : "+ DateUtil.now();
    }
}